from rest_framework_jwt.utils import jwt_payload_handler

#jwt 分为三部分数据  header  payload  signature

#自定义的响应体
#后台返回的响应体 里面只存在token  重新定义response 添加username，user_id
def jwt_response_payload_handler(token,user=None,request=None):
    return {
        'token':token,
        'username':user.username,
        'user_id':user.id
    }
#自定义载荷payload  原始的载荷之中缺乏mobile手机号这个变量
#故自定义进行操作
def meiduo_payload_handler(user):
    #里面有一些操作 username,user_id,exp,email
    payload = jwt_payload_handler(user)
    #删除邮箱
    del payload['email']

    #添加手机号
    payload['mobile'] = user.mobile
    return payload
